home *** CD-ROM | disk | FTP | other *** search
/ Komputer for Alle 2004 #2 / K-CD-2-2004.ISO / OpenOffice Sv / f_0397 / python-core-2.2.2 / lib / test / test_traceback.py < prev    next >
Encoding:
Python Source  |  2003-07-18  |  1.4 KB  |  47 lines
  1. """Test cases for traceback module"""
  2.  
  3. import unittest
  4. from test_support import run_unittest
  5.  
  6. import traceback
  7.  
  8. class TracebackCases(unittest.TestCase):
  9.     # For now, a very minimal set of tests.  I want to be sure that
  10.     # formatting of SyntaxErrors works based on changes for 2.1.
  11.  
  12.     def get_exception_format(self, func, exc):
  13.         try:
  14.             func()
  15.         except exc, value:
  16.             return traceback.format_exception_only(exc, value)
  17.         else:
  18.             raise ValueError, "call did not raise exception"
  19.  
  20.     def syntax_error_with_caret(self):
  21.         compile("def fact(x):\n\treturn x!\n", "?", "exec")
  22.  
  23.     def syntax_error_without_caret(self):
  24.         # XXX why doesn't compile raise the same traceback?
  25.         import badsyntax_nocaret
  26.  
  27.     def test_caret(self):
  28.         err = self.get_exception_format(self.syntax_error_with_caret,
  29.                                         SyntaxError)
  30.         self.assert_(len(err) == 4)
  31.         self.assert_("^" in err[2]) # third line has caret
  32.         self.assert_(err[1].strip() == "return x!")
  33.  
  34.     def test_nocaret(self):
  35.         err = self.get_exception_format(self.syntax_error_without_caret,
  36.                                         SyntaxError)
  37.         self.assert_(len(err) == 3)
  38.         self.assert_(err[1].strip() == "[x for x in x] = x")
  39.  
  40.  
  41. def test_main():
  42.     run_unittest(TracebackCases)
  43.  
  44.  
  45. if __name__ == "__main__":
  46.     test_main()
  47.